// test_env.c
#include <kunit/test.h>
#include <linux/module.h>
#include <linux/kthread.h>
#include <linux/completion.h>
#include "../hisi_l0_mem_pool.h"
#include "test_common.h"

static struct test_context g_test_ctx;

// Test initialization and cleanup functions
int l0_mem_pool_test_init(void *test) // Changed from struct kunit *test
{
    memset(&g_test_ctx, 0, sizeof(g_test_ctx));
    g_test_ctx.test_node_id = 0;
    return 0;
}
void l0_mem_pool_test_exit(void *test)
{
    // Cleanup will happen through normal l0_kfree calls
}

// Make this accessible to other test files
struct test_context *get_test_context(void)
{
    return &g_test_ctx;
}